# _*_ coding: UTF-8 _*_
# @Time     : 2021/11/2 10:36
# @Author   : Gong Yangyi
# @Site     : http://www.hzdledu.cn/
# @File     : w3s1_难点.py
# @Software : PyCharm
from selenium import webdriver
import time
from selenium.webdriver.common.keys import Keys


# 模拟同时按下多个按键
def foo1():
    driver = webdriver.Chrome()
    driver.maximize_window()

    driver.get('https://www.baidu.com/')

    driver.find_element_by_xpath('/html').send_keys(Keys.CONTROL, 'A')
    time.sleep(2)

    driver.quit()


# 验证码
# 1.要求开发去除验证码
# 2.要求提供万能验证码
# 3.可以使用一些OCR图像识别软件，但成功率不高

# 截图
def foo2():
    driver = webdriver.Chrome()
    driver.maximize_window()

    driver.get('https://www.baidu.com/')

    # 截取元素的截图，保存为png文件
    driver.find_element_by_id('su').screenshot('./baidu_button.png')

    try:
        driver.find_element_by_xpath('//*[@id="s-top-left"]/a[1]').click()
    except Exception as e:
        driver.get_screenshot_as_file('./baidu1.png')  # 截取整个网页保存为文件，文件是png
        raise Exception('点击新闻出现异常')
    finally:
        driver.quit()


# 执行js语句
# driver.execute_script
if __name__ == '__main__':
    # foo1()
    foo2()
